﻿// ------------------------------------------------------------------------
// 项目名称：Canroc.Net 
// 版权归属：Canroc（https://gitee.com/canroc）
//
// 许可证信息
// Canroc.Net项目主要遵循 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-APACHE 文件。
//
// 使用条款：
// 1.使用本项目应遵守相关法律法规和许可证的要求。
// 2.不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动。
// 3.任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任
//
// 免责声明
// 对于因使用本代码而产生的任何直接、间接、偶然、特殊或后果性损害，我们不承担任何责任。
//
// 其他重要信息
// Canroc.Net 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。
// ------------------------------------------------------------------------

using System.Security.Cryptography;

namespace Canroc.Net.Core.Extension;

/// <summary>
///     AES加密扩展
/// </summary>
public static class AESCryptoExtension
{
    /// <summary>
    ///     生成公钥
    /// </summary>
    /// <returns></returns>
    public static string GeneratePublicKey()
    {
        const string allowedChars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*_-";
        Random random = new();

        const int keyLength = 16;
        StringBuilder sb = new(keyLength);

        for (var i = 0; i < keyLength; i++)
        {
            var index = random.Next(allowedChars.Length);
            var c = allowedChars[index];
            sb.Append(c);
        }

        return sb.ToString();
    }

    /// <summary>
    ///     AES加密
    /// </summary>
    /// <param name="word">待加密字符串</param>
    /// <param name="key">密钥</param>
    /// <param name="iv">偏移量</param>
    /// <returns></returns>
    public static string AESEncryption(this string word, string key, string iv)
    {
        try
        {
            if (word.IsNullOrEmpty() || key.IsNullOrEmpty() || iv.IsNullOrEmpty())
            {
                return "";
            }

            using var aes = Aes.Create();
            aes.Key = Encoding.UTF8.GetBytes(key);
            aes.IV = Encoding.UTF8.GetBytes(iv);
            aes.Mode = CipherMode.CBC;
            aes.Padding = PaddingMode.PKCS7;

            using var encryptor = aes.CreateEncryptor();
            var wordBytes = Encoding.UTF8.GetBytes(word);
            var encryptedData = encryptor.TransformFinalBlock(wordBytes, 0, wordBytes.Length);

            return Convert.ToBase64String(encryptedData);
        }
        catch (Exception)
        {
            return "";
        }
    }

    /// <summary>
    ///     AES解密
    /// </summary>
    /// <param name="word">待加密字符串</param>
    /// <param name="key">密钥</param>
    /// <param name="iv">偏移量</param>
    /// <returns></returns>
    public static string AESDecrypt(this string word, string key, string iv)
    {
        try
        {
            if (word.IsNullOrEmpty() || key.IsNullOrEmpty() || iv.IsNullOrEmpty())
            {
                return "";
            }

            using var aes = Aes.Create();
            aes.Key = Encoding.UTF8.GetBytes(key);
            aes.IV = Encoding.UTF8.GetBytes(iv);
            aes.Mode = CipherMode.CBC;
            aes.Padding = PaddingMode.PKCS7;

            using var decrypted = aes.CreateDecryptor();
            var wordBytes = Convert.FromBase64String(word);
            var decryptedData = decrypted.TransformFinalBlock(wordBytes, 0, wordBytes.Length);

            return Encoding.UTF8.GetString(decryptedData);
        }
        catch (Exception)
        {
            return "";
        }
    }
}